package middleware

import (
	"github.com/gin-gonic/gin"
	"partyus_app/response"
)

func CheckGeneralRole() gin.HandlerFunc {
	return func(c *gin.Context) {
		role, exists := c.Get("role")
		if !exists {
			response.CommonResp(c, 1, "权限不足", nil)
			c.Abort()
			return
		}
		// 假设 0 是区团工委，1 是一般团工委
		if role == 0 || role == 1 {
			c.Next() // 满足任意角色要求，继续处理请求
			return
		}
		// 角色不满足，返回权限不足
		response.CommonResp(c, 1, "权限不足", nil)
		c.Abort()
	}
}

func CheckDistrictRole() gin.HandlerFunc {
	return func(c *gin.Context) {
		role, exists := c.Get("role")
		if !exists || role != 0 { // 假设 0 表示区团工委
			response.CommonResp(c, 1, "权限不足", nil)
			c.Abort()
			return
		}
		c.Next()
	}
}
